Python. list.pop() based on index